Added comments to group code by problem