Model changes for stats task