Task 4 now passes the test