Added two train test, fixed bug in load checking for null load in train