Create solve.py