Submitted Assignment 2 - Part 4 - Predict