Minor changes to plotting code.
Added plotting function. It's not too fancy but it works.
Merge pull request #6 from fercol/master

plotting...