update for train style