Improve matplotlib plot

Add legend, limits, title and major y grids.
Implements #23