add stats tracking