Add the widget behavior