Clean up cell class