updated highlight code