Corrected one of the test cases and added another one.