solutions/prob1.py