Update code to display data