Add some things for back propagation