Changed the data to be held in a treemap