fixed error, added more to the test class