add code highlighting stuff back in