Got the tests to work! YAY