Back propagation works now