Add classes to cells for testing