added passing test for display function