Some minor cleaning up.
Some work on the plotting of graphs (#31).