Add stat updating