Adding tests for widgets