diff --git a/hw5/countD.py b/hw5/countD.py
index b8b4c06..75c68c0 100644
--- a/hw5/countD.py
+++ b/hw5/countD.py
@@ -1,3 +1,6 @@
+from core import *
+from ooc import *
+
 """
 Get the dataset first, download title.csv put it in the pa1 folder
 https://www.dropbox.com/s/zl7yt8cl0lvajxg/title.csv?dl=0