Add figure for EDI