Added all 6 test cases to the class