Added update_selection method