Add counter function