add figure for partition example