Added initialization of train orienration in InitializeTestEnvironment procedure
Corrected tests to avoid errors in the model