Adding files to train and test the model
Commiting program modifications
Modified the README file