Trying to extend backprop to multi layer