updated for train