minor points: I would prefer it to be imported as `_solve` -- changing the name makes it sound like too different of a function IMO. And there's probably no need to call the solution `sol_five`; `sol` is fine.