Added callback parameter to add_widget