Update prime.rb