Update README.md, improve solve.py interface