Add documentation to the code