add docs to code