Updated unit test answers