Prediction Assignment writeup