Update stat_generator.rb