clean up
clean up tests, add some docs