[Issue #86] clean up tests and doc string