Okay, add get_stats func