Part 3: Training the Neural Network