Add count feature