Mohit | Added test cases