Improved VIM for python