test.rb

class Pentagon

  def initialize(element)
    penta = []
    for i in 1..element
      penta << (i * (3 * i - 1) / 2)
      i += 1
    end

    puts penta
  end

end

pe = Pentagon.new(10)