diff --git a/project/app/models.py b/project/app/models.py index 4237a9e..3638757 100644 --- a/project/app/models.py +++ b/project/app/models.py @@ -51,3 +51,5 @@ class Roster: def reset_roster(self): """Reset students to their original list (like at the start of the day).""" self.students = self.original_students[:] + + diff --git a/project/app/views.py b/project/app/views.py index c457f66..5b23479 100644 --- a/project/app/views.py +++ b/project/app/views.py @@ -43,4 +43,8 @@ print("Remaining today:", roster.students) roster.reset_roster() print("Next day reset:", roster.students) +remove = roster.remove_student +if not removed: + print ("Student not Found.") +