Update solution3-1.py