made some changes
adding test cases