Added models for trainschedule