bug fixes
update docs