Added some widget code for testing