code cleanup
update docs