Modify plot-image.py for the New Design