Update python.vim

highlight python function