Add cleanup method in output class

diff --git a/image_creator/output/__init__.py b/image_creator/output/__init__.py
index 7143785..3087c9c 100644 (file)
--- a/image_creator/output/__init__.py
+++ b/image_creator/output/__init__.py
@@ -45,6 +45,9 @@ class Output(object):
     def output(self, msg='', new_line=True):
         pass
 
+    def cleanup(self):
+        pass
+
     def _get_progress(self):
         progress = self._Progress
         progress.output = self