Fix syntax highlighting!
(It's still not quite 100%, but it'll work for me)