Wrote the test cases