Update python script for algorithms without original images, e.g. function plots