update docstring for checkgrad