Made all of the test case in class work