Implemented 2 test cases for addition and removal of classes