Added comments to problem 2 and problem 3