diff --git a/setup.py b/setup.py
index 60bef7d..8369721 100644
--- a/setup.py
+++ b/setup.py
@@ -17,7 +17,7 @@
 
 setup(
     name='tutormagic',
-    version='0.2.1',
+    version='0.3.1',
     description='Magic to display pythontutor.com from a code cell in the IPython notebook.',
     long_description=long_description,
     url='https://github.com/kikocorreoso/tutormagic',
@@ -39,6 +39,10 @@
         'Programming Language :: Python :: 3.3',
         'Programming Language :: Python :: 3.4',
         'Programming Language :: Python :: 3.5',
+        'Programming Language :: Python :: 3.6',
+        'Programming Language :: Python :: 3.7',
+        'Programming Language :: Python :: 3.8',
+        'Programming Language :: Python :: 3.9',
     ],
 
     # What does your project relate to?
diff --git a/tutormagic.py b/tutormagic.py
index 8780f38..7ce4255 100644
--- a/tutormagic.py
+++ b/tutormagic.py
@@ -16,7 +16,7 @@
 
 """
 
-__version__="0.3.0"
+__version__="0.3.1"
 
 #-----------------------------------------------------------------------------
 # Copyright (C) 2015-2017 Kiko Correoso and the pythontutor.com developers