Update .vimrc

I actually needed that function I removed.