Update solution2.py