Fix code colouring