Minor fixes for trains, use IsEntityValid for other calls checking if the entity exists

Minor fixes for trains, use IsEntityValid for other calls checking if the entity exists #531