Add setStats method