Change unet method