cleaned up some code and made the test pass