improved documentation -- TODO add link to repo and add link to API docs