changed homework 1 test code to match original expectation