Completed assignment 5 neural networks