added code for displaying models