Add support for both python2 and python3

[tom025]

I have tested this using vim with python3 on macos.

This would be really useful if it was merged @jaxbot. Also thank you @CandySunPlus.