Initial commit for prediction algorithm