diff --git a/examples/Example11_Synapses.net.nml b/examples/Example11_Synapses.net.nml
index c2571ab..9dea1da 100644
--- a/examples/Example11_Synapses.net.nml
+++ b/examples/Example11_Synapses.net.nml
@@ -1,4 +1,4 @@
-<neuroml xmlns="http://www.neuroml.org/schema/neuroml2"  xmlns:xs="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.neuroml.org/schema/neuroml2 https://raw.github.com/NeuroML/NeuroML2/development/Schemas/NeuroML2/NeuroML_v2.3.xsd" id="Example11_Synapses">
+<neuroml xmlns="http://www.neuroml.org/schema/neuroml2"  xmlns:xs="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.neuroml.org/schema/neuroml2 https://raw.github.com/NeuroML/NeuroML2/development/Schemas/NeuroML2/NeuroML_v2.3.1.xsd" id="Example11_Synapses">
     <notes>Generated by NeuroMLlite v0.5.9
     Generated network: Example11_Synapses
     Generation seed: 1234
diff --git a/examples/Example2_TestNetwork.net.nml b/examples/Example2_TestNetwork.net.nml
index 56c0694..d6a694e 100644
--- a/examples/Example2_TestNetwork.net.nml
+++ b/examples/Example2_TestNetwork.net.nml
@@ -1,4 +1,4 @@
-<neuroml xmlns="http://www.neuroml.org/schema/neuroml2"  xmlns:xs="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.neuroml.org/schema/neuroml2 https://raw.github.com/NeuroML/NeuroML2/development/Schemas/NeuroML2/NeuroML_v2.3.xsd" id="Example2_TestNetwork">
+<neuroml xmlns="http://www.neuroml.org/schema/neuroml2"  xmlns:xs="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.neuroml.org/schema/neuroml2 https://raw.github.com/NeuroML/NeuroML2/development/Schemas/NeuroML2/NeuroML_v2.3.1.xsd" id="Example2_TestNetwork">
     <notes>Generated by NeuroMLlite v0.5.9
     Generated network: Example2_TestNetwork
     Generation seed: 1234</notes>
diff --git a/examples/Example3_Network.net.nml b/examples/Example3_Network.net.nml
index 8cb1d6b..fe8332c 100644
--- a/examples/Example3_Network.net.nml
+++ b/examples/Example3_Network.net.nml
@@ -1,4 +1,4 @@
-<neuroml xmlns="http://www.neuroml.org/schema/neuroml2"  xmlns:xs="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.neuroml.org/schema/neuroml2 https://raw.github.com/NeuroML/NeuroML2/development/Schemas/NeuroML2/NeuroML_v2.3.xsd" id="Example3_Network">
+<neuroml xmlns="http://www.neuroml.org/schema/neuroml2"  xmlns:xs="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.neuroml.org/schema/neuroml2 https://raw.github.com/NeuroML/NeuroML2/development/Schemas/NeuroML2/NeuroML_v2.3.1.xsd" id="Example3_Network">
     <notes>Generated by NeuroMLlite v0.5.9
     Generated network: Example3_Network
     Generation seed: 1234</notes>
diff --git a/examples/Example4_PyNN.net.nml b/examples/Example4_PyNN.net.nml
index 0869fd4..82747ce 100644
--- a/examples/Example4_PyNN.net.nml
+++ b/examples/Example4_PyNN.net.nml
@@ -1,4 +1,4 @@
-<neuroml xmlns="http://www.neuroml.org/schema/neuroml2"  xmlns:xs="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.neuroml.org/schema/neuroml2 https://raw.github.com/NeuroML/NeuroML2/development/Schemas/NeuroML2/NeuroML_v2.3.xsd" id="Example4_PyNN">
+<neuroml xmlns="http://www.neuroml.org/schema/neuroml2"  xmlns:xs="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.neuroml.org/schema/neuroml2 https://raw.github.com/NeuroML/NeuroML2/development/Schemas/NeuroML2/NeuroML_v2.3.1.xsd" id="Example4_PyNN">
     <notes>Generated by NeuroMLlite v0.5.9
     Generated network: Example4_PyNN
     Generation seed: 1234
diff --git a/examples/Example6_PyNN.net.nml b/examples/Example6_PyNN.net.nml
index 8692b8f..09302c1 100644
--- a/examples/Example6_PyNN.net.nml
+++ b/examples/Example6_PyNN.net.nml
@@ -1,4 +1,4 @@
-<neuroml xmlns="http://www.neuroml.org/schema/neuroml2"  xmlns:xs="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.neuroml.org/schema/neuroml2 https://raw.github.com/NeuroML/NeuroML2/development/Schemas/NeuroML2/NeuroML_v2.3.xsd" id="Example6_PyNN">
+<neuroml xmlns="http://www.neuroml.org/schema/neuroml2"  xmlns:xs="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.neuroml.org/schema/neuroml2 https://raw.github.com/NeuroML/NeuroML2/development/Schemas/NeuroML2/NeuroML_v2.3.1.xsd" id="Example6_PyNN">
     <notes>Generated by NeuroMLlite v0.5.9
     Generated network: Example6_PyNN
     Generation seed: 1234
diff --git a/examples/Example6_PyNN_noinputs.net.nml b/examples/Example6_PyNN_noinputs.net.nml
index 36041b0..12be79b 100644
--- a/examples/Example6_PyNN_noinputs.net.nml
+++ b/examples/Example6_PyNN_noinputs.net.nml
@@ -1,4 +1,4 @@
-<neuroml xmlns="http://www.neuroml.org/schema/neuroml2"  xmlns:xs="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.neuroml.org/schema/neuroml2 https://raw.github.com/NeuroML/NeuroML2/development/Schemas/NeuroML2/NeuroML_v2.3.xsd" id="Example6_PyNN_noinputs">
+<neuroml xmlns="http://www.neuroml.org/schema/neuroml2"  xmlns:xs="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.neuroml.org/schema/neuroml2 https://raw.github.com/NeuroML/NeuroML2/development/Schemas/NeuroML2/NeuroML_v2.3.1.xsd" id="Example6_PyNN_noinputs">
     <notes>Generated by NeuroMLlite v0.5.9
     Generated network: Example6_PyNN_noinputs
     Generation seed: 1234
diff --git a/examples/Example7_Brunel2000.net.nml b/examples/Example7_Brunel2000.net.nml
index 34d24d2..c7e6757 100644
--- a/examples/Example7_Brunel2000.net.nml
+++ b/examples/Example7_Brunel2000.net.nml
@@ -1,4 +1,4 @@
-<neuroml xmlns="http://www.neuroml.org/schema/neuroml2"  xmlns:xs="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.neuroml.org/schema/neuroml2 https://raw.github.com/NeuroML/NeuroML2/development/Schemas/NeuroML2/NeuroML_v2.3.xsd" id="Example7_Brunel2000">
+<neuroml xmlns="http://www.neuroml.org/schema/neuroml2"  xmlns:xs="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.neuroml.org/schema/neuroml2 https://raw.github.com/NeuroML/NeuroML2/development/Schemas/NeuroML2/NeuroML_v2.3.1.xsd" id="Example7_Brunel2000">
     <notes>Generated by NeuroMLlite v0.5.9
     Generated network: Example7_Brunel2000
     Generation seed: 1234
diff --git a/examples/LEMS_SimExample3.xml b/examples/LEMS_SimExample3.xml
index 5204c5b..7e00601 100644
--- a/examples/LEMS_SimExample3.xml
+++ b/examples/LEMS_SimExample3.xml
@@ -2,7 +2,7 @@
 
     <!-- 
 
-        This LEMS file has been automatically generated using PyNeuroML v1.2.13 (libNeuroML v0.5.9)
+        This LEMS file has been automatically generated using PyNeuroML v1.3.0 (libNeuroML v0.6.0)
 
      -->
 
diff --git a/examples/LEMS_SimExample4.xml b/examples/LEMS_SimExample4.xml
index e584554..be8366a 100644
--- a/examples/LEMS_SimExample4.xml
+++ b/examples/LEMS_SimExample4.xml
@@ -2,7 +2,7 @@
 
     <!-- 
 
-        This LEMS file has been automatically generated using PyNeuroML v1.2.13 (libNeuroML v0.5.9)
+        This LEMS file has been automatically generated using PyNeuroML v1.3.0 (libNeuroML v0.6.0)
 
      -->
 
diff --git a/neuromllite/__init__.py b/neuromllite/__init__.py
index 0c96744..6e0c076 100644
--- a/neuromllite/__init__.py
+++ b/neuromllite/__init__.py
@@ -1,6 +1,6 @@
 import collections
 
-__version__ = "0.5.9"
+__version__ = "0.6.0"
 
 # import pyNN
 # import nest
diff --git a/neuromllite/gui/NMLliteUI.py b/neuromllite/gui/NMLliteUI.py
index 389b9b2..deb8d92 100644
--- a/neuromllite/gui/NMLliteUI.py
+++ b/neuromllite/gui/NMLliteUI.py
@@ -1044,7 +1044,7 @@ def replotSimResults(self):
         ax_heatmap.clear()
 
         if len(heat_array) > 0:
-            cm = matplotlib.cm.get_cmap("jet")
+            cm = matplotlib.pyplot.get_cmap("jet")
             hm = ax_heatmap.pcolormesh(heat_array, cmap=cm)
             # cbar = ax_heatmap.colorbar(im)