add tests
still working