Add some comments to problem 1 solution
Comments for problem 2