4 tests passing in part 1
make all tests pass in part 1