update docs
bug fix
clean up; fixing
added more tests