Improving code syntax highlighting a little bit

Just improving code syntax highlighting a little bit.