modify get_stats func