At this line, shouldn't `best_img` be returned instead of `gen_img`? https://github.com/adamian98/pulse/blob/40cacb97df28081e0edf8440a8c3f7c4e6205732/PULSE.py#L177