Answer to #3 unit test