Add the ability to choose what to display