Re-add code highlighting