added comments to Problem 2