Add documentation to source code