Cleaned up predictions and created Prediction class.