[Refac] started to test backprop