polished highlight code so it works on UG