Update docstring + Use np.diff