Changing np.round to np.rint