Python Solve Function