updated my unit test some