Add python styles for vim