Add count reset functionality