Add Python syntax highlighting