Started to work on all the classes that we need for prediction. Not working yet.