Made a few adjustments:
1) Improved solve() call
2) Removed an unneded print
3) improved documentation