adds tests
update tests and setup