First stab at the prediction assignment