fixed the gradients
getting there