Fix linter: extend used outside of class or module

main.rb

@@ -1,11 +1,22 @@
 require_relative 'school_library/app'
-require_relative 'school_library/modules/display_menu'
 def main
   App.run
 end
 
+def display_menu
+  puts "\nPlease choose an option by enter a number:"
+  puts '1. List all books'
+  puts '2. List all people'
+  puts '3. Create a person'
+  puts '4. Create a book'
+  puts '5. Create a rental'
+  puts '6. List all rentals for a given person (by ID)'
+  puts '7. Exit'
+  print 'Enter your choice: '
+  gets.chomp.to_i
+end
+
 def handle_choice(choice)
-  extend DisplayMenu
   case choice
   when 1
     list_books

school_library/app.rb

@@ -1,20 +1,14 @@
-require_relative 'modules/display_menu'
 require_relative 'modules/create_persons'
 require_relative 'modules/create_books'
 require_relative 'modules/lists'
 
 class App
-  extend DisplayMenu
   extend CreatePerson
   extend CreateBook
   extend Lists
 
   def self.run
-    loop do
-      display_menu
-      choice = gets.chomp.to_i
-      handle_choice(choice)
-      break if choice == 7
-    end
+    choice = handle_choice(choice) while choice != 7
+    puts 'Thanks for using the School Library!'
   end
 end