Update RGB_restore.py