docs and clean up tests a bit