add gradient check