Added highlight code feature