Python functionality improved in vim
Small vim change