clean up documentation, continue writing unit tests