trying to add save_images.py file