diff --git a/mnists/__init__.py b/mnists/__init__.py
index e0a4fe8..268013f 100644
--- a/mnists/__init__.py
+++ b/mnists/__init__.py
@@ -3,7 +3,7 @@
 # Copyright (C) 2018 Connor Olding
 # Distributed under terms of the MIT license.
 
-__version__ = "0.3.2"
+__version__ = "0.4.0"
 
 import array
 import gzip
diff --git a/setup.py b/setup.py
index 2292d5d..9879f24 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@ from setuptools import setup
 
 setup(
     name='mnists',
-    version='0.3.2',
+    version='0.4.0',
     packages=[
         'mnists',
     ],