Moved stats to model instead of unit