Add class Train with initialize method.