Add solution class

[BillyTheKidPhysicist]

Current implementation returns a dictionary. A more user friendly feature would be to return a solution class that can plot results and return useful parameters.