Add more gradients

Fixes #301