Update figures on gradient computation