Adding comments to assignment code