Added Train class
Added Train class