Submitted Assignment 3 - Part 4 - Neural network prediction function