Update soln1.py