Adding the back prop to the traininer
adding back prop