[Testing] got backprop test to pass