Example code for CSE 5526: Neural Networks