Change class in highlight script.