fixed highlighting issue back in class #15