Worked on getting the tests to work