Add better syntax highlighting