Fixed bug with initializing predictor
Added training data