I added a utility file for reusability. train, test, evaluate scripts are much smaller now and easier to read