self._display.topic_table.attach in collection

2 files changed, 4 insertions, 2 deletions

diff --git a/ReckonPrimer.activity/collection.py b/ReckonPrimer.activity/collection.py
index 787fc4b..a748cb0 100644
--- a/ReckonPrimer.activity/collection.py
+++ b/ReckonPrimer.activity/collection.py
@@ -15,7 +15,10 @@ class Collection:
     def define_coll_gui(self):
         """ Define gui-elements for presenting the collcetion. """
         self.topic_box = gtk.HBox(True, 0)
-        self.topic_table.attach(self.topic_box, 0, 6, 0, 1)
+        self._display.topic_table.attach(self.topic_box, 0, 6, 0, 1)
+
+    def set_coll_gui(self):
+        """ Set gui-elements according to Collection.data. """
         _i = 0
         for _t in topics:
             _i = _i + 1
diff --git a/ReckonPrimer.activity/display.py b/ReckonPrimer.activity/display.py
index 4f1f215..7d0b577 100755
--- a/ReckonPrimer.activity/display.py
+++ b/ReckonPrimer.activity/display.py
@@ -388,7 +388,6 @@ class Display:
         self.correct_count = self.correct_count + 1
         self.correct_counter.set_text(str(self.correct_count))
         self.display_table.destroy()
-
         
     def offer_coll_to_learner(self, topics):
         """TODO: get the users choice from buttons above the settings"""