Changed solve.py to be more user-friendly