added more test cases to increase coverage