Added more models and api call for stats + UI