Add count function