clean up and fix docs