[change] generic add/drop unit; [add] impls for menu class

1 files changed, 5 insertions, 2 deletions

diff --git a/src/wheelwork.lisp b/src/wheelwork.lisp
index e411c52..154adaa 100644
--- a/src/wheelwork.lisp
+++ b/src/wheelwork.lisp
@@ -5,13 +5,16 @@
 (defvar *application* nil
   "current application")
 
-(defun add-unit (unit)
+(defgeneric add-unit (unit))
+
+(defmethod add-unit ((unit unit))
   "Adds a unit to the display."
   (assert *application*)
   (push unit (application-scene *application*))
   (setf (unit-in-scene-p unit) t))
 
-(defun drop-unit (unit)
+(defgeneric drop-unit (unit))
+(defmethod drop-unit ((unit unit))
   "A removes a unit from the display."
   (assert *application*)
   (setf (application-scene *application*)