small improvements to Python image plot formatting.
Merge pull request #3 from msarahan/master

Improve Python image plotting