Conditionally render a class to a grid cell, then remove it on a state change

Question

Ok, so first time question-asker here, but I\'m having trouble rendering a class to a grid cell, then removing it. The idea is the grid is the floor, and the class represent