problem2.py now saves to proper png file