Update prime_shifting.rb