Small updates to figure plotting
Merge pull request #5 from gbledt/master

Small updates to figure plotting