Final push - add comments to codes