Add python mode for vim