Use np.sum, not vanilla sum