Add test for widgets