create class Train with initialize method