Improve gradient features to use .reset()