Fix code hilighting so it's easier to read