added train to the update