diff --git a/src/.config/nvim/init.vim b/src/.config/nvim/init.vim
index 57f9543..ba3588e 100755
--- a/src/.config/nvim/init.vim
+++ b/src/.config/nvim/init.vim
@@ -138,3 +138,8 @@ require'nvim-treesitter.configs'.setup {
   },
 }
 EOF
+
+" Code folding
+set foldmethod=expr
+set foldexpr=nvim_treesitter#foldexpr()
+autocmd BufEnter * normal zR