Add documentation for the code