modify code for fizzbuzz.rb