Updated treemap UI