added search for treemap as well