Update comments problem1.py