Customize python syntax highlighting