I think I nailed the gradients this time :)

I think I nailed the gradients this time :)