add support for PU jet ID