Fixed issues with backprop -- now passing gradient checking