FIX use np.bincount