started adding unit test cases