Checking for Prime Number

prime.rb

def prime? n
  return true if n == 2
  return false if n < 2 or n % 2 == 0
  for d in (3..Math.sqrt(n).to_i).step(2) do
    return false if n % d == 0
  end
  return true
end