-adding new classes
-creating enum
-changing buttons for grid