highlight code correctly