Tweaks to backprop