got two of the tests to pass