remade unit train menu for later ui improvements