improvement to add count