fixing display_images (small change)