Update sol.py