clean up and update docs