added all the code and train+test data