Added tests for widgets and display