Adding few more tests
Adding more test cases and assertions