Add stats updating