Tested two modules: load_train and load_test