Add tree map option