Update solution1-2.py