Simplify the method to update the counts