Improvements on TreeView.

Signed-off-by: Arica Liu <aricaliu@wpi.edu>