problem 1a.py