Add stats code