Add description to problem1.py