added comments to Problem 1 solution