modify psnr func