Update ImageCreation.py