added proper testing
updated docs