Initialize train class