Improving code syntax highlighting a little bit