add tests, remove prints