Got the tests working, but not passing