Removed treemap logic