Add functionality to display table as well as tree.