Fixed bug with the lack of highlighting cell after step.