Add count for token usage (#1)