Update snr_image.py