bug fixes, docs update