Add comments to problem 2 code
Add comments to problem 1 code
Add comments to problem 3 code