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

Edited by Fabian Nuraddin Alexander Gabel