Fixes up code highlighting