Fixed saving images
Updated for PyQt5