Update solution.py