First version of Train class completed