diff --git a/dnc/__init__.py b/dnc/__init__.py
index 5f7ce86..0ee2997 100644
--- a/dnc/__init__.py
+++ b/dnc/__init__.py
@@ -1 +1,3 @@
-#!/usr/bin/env python3
\ No newline at end of file
+#!/usr/bin/env python3
+
+from .dnc import DNC
diff --git a/dnc/dnc.py b/dnc/dnc.py
index 7faf045..f303a61 100644
--- a/dnc/dnc.py
+++ b/dnc/dnc.py
@@ -9,8 +9,8 @@ from torch.nn.utils.rnn import pad_packed_sequence as pad
 from torch.nn.utils.rnn import pack_padded_sequence as pack
 from torch.nn.utils.rnn import PackedSequence
 
-from util import *
-from memory import *
+from .util import *
+from .memory import *
 
 
 class DNC(nn.Module):
diff --git a/dnc/memory.py b/dnc/memory.py
index f41a11c..764cfb8 100644
--- a/dnc/memory.py
+++ b/dnc/memory.py
@@ -6,7 +6,7 @@ from torch.autograd import Variable as var
 import torch.nn.functional as F
 import numpy as np
 
-from util import *
+from .util import *
 
 
 class Memory(nn.Module):