diff --git a/student.rb b/student.rb index 514cc09..1ecf650 100644 --- a/student.rb +++ b/student.rb @@ -10,6 +10,10 @@ def initialize(first_name, last_name, grade) def senior? grade == 12 end + + def junior? + grade == 11 + end def to_s "#{last_name}, #{first_name}" @@ -20,11 +24,21 @@ def seniors(students) students.select { |student| student.senior? } end +def juniors(students) + students.select { |student| student.junior? } +end + fred = Student.new("Fred", "James", 12) sarah = Student.new("Sarah", "Smith", 12) jack = Student.new("Jack", "Gong", 11) all_students = [fred, sarah, jack] +# Added some formatting for readibility +puts 'Seniors:' seniors(all_students).each do |student| - puts student + puts " * #{student}" +end +puts 'Juniors:' +juniors(all_students).each do |student| + puts " * #{student}" end