design changes. Got rid of a bunch of unneeded gradients