Fix code highlighting to look better.