final fixes for image plotting