Correction in preprocessing.py for saving niftii file