Got the tests to pass