diff --git a/node2vec/__init__.py b/node2vec/__init__.py
index fb3d1a0..4b262c3 100644
--- a/node2vec/__init__.py
+++ b/node2vec/__init__.py
@@ -1,4 +1,4 @@
-from .node2vec import Node2Vec
 from . import edges
+from .node2vec import Node2Vec
 
-__version__ = '0.3.3'
\ No newline at end of file
+__version__ = '0.4.0'
diff --git a/setup.py b/setup.py
index 9acfa62..5010285 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 setup(
     name='node2vec',
     packages=['node2vec'],
-    version='0.3.3',
+    version='0.4.0',
     description='Implementation of the node2vec algorithm.',
     author='Elior Cohen',
     author_email='elior.cohen.p@gmail.com',