Improve code highlight style