Resolve Display issue

Change return in display()