polish code; continue writing documentation