Update solution3-2.py