Implemented two more test cases