Function on_activate_trains 

pub fn on_activate_trains(
    _: On<'_, '_, Activate>,
    cmd: Commands<'_, '_>,
    q_trains: Query<'_, '_, Entity, With<Train>>,
    curr_state: Res<'_, State<UiState>>,
    next_state: ResMut<'_, NextState<UiState>>,
)