Added two more test cases as per review.