Completed the second unit test case