diff --git a/6-functions/33_drive_thru.py b/6-functions/33_drive_thru.py
index 04a1aa3..7814c15 100644
--- a/6-functions/33_drive_thru.py
+++ b/6-functions/33_drive_thru.py
@@ -1,30 +1,19 @@
 # Drive-Thru 🚙
 # Codédex
 
-def get_item(x):
-  if x == 1:
-    return '🍔 Cheeseburger'
-  elif x == 2:
-    return '🍟 Fries'
-  elif x == 3:
-    return '🥤 Soda'
-  elif x == 4:
-    return '🍦 Ice Cream'
-  elif x == 5:
-    return '🍪 Cookie'
-  else:
-    return "invalid option"
+menu = ['Cheeseburger', 'Fries', 'Soda', 'Ice Cream', 'Cookie']
+
+def get_item(i):
+  return(menu[i - 1])
 
 def welcome():
-  print('Welcome to Sonnyboy\'s Diner!')
-  print('Here\'s the menu:')
-  print('1. 🍔 Cheeseburger')
-  print('2. 🍟 Fries')
-  print('3. 🥤 Soda')
-  print('4. 🍦 Ice Cream')
-  print('5. 🍪 Cookie')
+  print("Welcome! Here is our menu:")
+  print(menu)
 
 welcome()
 
-option = int(input('What would you like to order? '))
-print(get_item(option))
+order = int(input("What would you like to order? "))
+if order > 5:
+  print("Invalid order.")
+else:
+  print(get_item(order))