Added a way to save results. Input images are converted to floats