What should I do with my Treemap?

Hey guys, I'm a CS student at U of T and for an assignment were doing this http://www.cdf.utoronto.ca/~csc148h/winter/assignments/a2/. That's where you come in Reddit, what cool things would you like to see implemented in for enhancements?

TL;DR: Im doing a Treemap for CS class, any suggestions on cool enhancements ?