Add documentation from the code walkthrough