Added comments to HW 02