add homework changes

added my changes to make the tests work.