Add few comments to the 2nd homework solution