added a few comments to the final homework