fixes in back_prop_step