add tests now work