Adding comments to problem1.py and problem2.py