diff --git a/init.lua b/init.lua
index 12772f6..0cf801f 100644
--- a/init.lua
+++ b/init.lua
@@ -7,7 +7,7 @@ vim.opt.shiftwidth = 2          -- Indent width
 vim.opt.expandtab = true        -- Use spaces instead of tabs
 vim.opt.wrap = true
 
-vim.cmd.colorscheme("retrobox")
+-- vim.cmd.colorscheme("retrobox")
 vim.opt.termguicolors = true
 
 vim.opt.clipboard = 'unnamedplus' -- System clipboard
@@ -61,3 +61,7 @@ require("rakshit")
 
 vim.cmd("packadd nvim.undotree")
 vim.keymap.set("n", "<leader>u", require("undotree").open)
+
+vim.keymap.set("n", "<leader>ee", function ()
+  vim.cmd("NvimTreeFocus")
+end)
diff --git a/lua/rakshit/plugins/filemanager.lua b/lua/rakshit/plugins/filemanager.lua
index 33bf771..6eeb33c 100644
--- a/lua/rakshit/plugins/filemanager.lua
+++ b/lua/rakshit/plugins/filemanager.lua
@@ -25,9 +25,15 @@ require("nvim-tree").setup({
     width = 30,
   },
 
+  on_attach = function(bufnr)
+    local api = require("nvim-tree.api")
+    api.config.mappings.default_on_attach(bufnr)
+    vim.keymap.set("n", "<Esc>", api.tree.close, { buffer = bufnr, nowait = true })
+  end,
+
   actions = {
     open_file = {
-      quit_on_open = true, -- closes tree when opening a file
+      quit_on_open = true,
     },
   },