Add better python syntax highlighting