Improve figures
Merge pull request #73 from ahojnnes/fig

Improve figures