Add stats collection