Add logic to set class on active cells.