Added test for display_widget, but needs additional support for widgets