Add a system to update the unit selector on click.