Add code to display diagram parts