Shallow Neural Network -- 2

I am implementing a Shallow Neural Network. I have doubts anbd not sure how to exactly implement Stochastic Gradient Descent for Mnistdataset. Kindly advise on the code and improvements for implementing the full Neural Network.