Wrote some test cases