Add BUNIT keyword to output FITS files to specify units