Update to gradient arrows plot