updated vim script. Open a new python file and do the trick