clean up
update docs