First set of test cases working