Move highlight rendering code into drawProblem()