add figure + correction