Update main.py

Author: PyCoder

share/kwipe/main.py

@@ -224,6 +224,7 @@ def on_cancel(self):
                                  self.control_list[device][0].current_round,
                                  self.control_list[device][0].position,
                                  self.control_list[device][0].offset)
+        self.create_device_tree()
 
     def on_clear(self):
         rows = self.workTableWidget.rowCount()
@@ -247,6 +248,7 @@ def on_finish(self):
                     serial = utils.get_partition_info(k)[1]
                     self.remove_status(serial)
                 del self.control_list[k]
+        self.create_device_tree()
 
     def create_action_menu(self):
         # TreeView Menu