Update widget_selection.py