Add documentation to code