solution.py