diff --git a/main.py b/main.py index 7e41e92..d0a6eaf 100644 --- a/main.py +++ b/main.py @@ -34,6 +34,8 @@ break if option == 1: myOrder.add_lemonade() + if option == 2: + myOrder.add_coke() if option == 3: myOrder.add_coffee() if option == 5: diff --git a/order.py b/order.py index 1b64414..d5fe1a7 100644 --- a/order.py +++ b/order.py @@ -38,3 +38,8 @@ def add_shake(self): self.total_cost += 6 self.items.append("shake") print("Added shake") + + def add_coke(self): + self.total_cost += 10 + self.items.append("coke") + print("Added coke")