Merge pull request #3 from msarahan/master

Improve Python image plotting