Added java class that uses the model to train the data