added np.array rather than array in velocity_disp