Added test cases to the tests