Index: /issm/trunk-jpl/src/m/plot/plot_unit.py
===================================================================
--- /issm/trunk-jpl/src/m/plot/plot_unit.py	(revision 21425)
+++ /issm/trunk-jpl/src/m/plot/plot_unit.py	(revision 21426)
@@ -29,5 +29,5 @@
 	# {{{ define wich colormap to use 
 	try:
-		defaultmap=plt.cm.get_cmap('viridis')
+		defaultmap=plt.cm.get_cmap('viridis',colorlevels)
 	except AttributeError:
 		print("Viridis can't be found (probably too old Matplotlib) reverting to gnuplot colormap")
Index: /issm/trunk-jpl/src/m/plot/plotmodel.py
===================================================================
--- /issm/trunk-jpl/src/m/plot/plotmodel.py	(revision 21425)
+++ /issm/trunk-jpl/src/m/plot/plotmodel.py	(revision 21426)
@@ -6,5 +6,5 @@
 	import pylab as p
 	import matplotlib.pyplot as plt
-	from mpl_toolkits.axes_grid1 import ImageGrid, AxesGrid
+        from mpl_toolkits.axes_grid1 import ImageGrid, AxesGrid
 except ImportError:
 	print "could not import pylab, matplotlib has not been installed, no plotting capabilities enabled"
@@ -17,5 +17,5 @@
 	'''
 
-	#First process options 
+	#First process options
 	options=plotoptions(*args)
 
@@ -36,5 +36,5 @@
 		nrows=np.ceil(numberofplots/subplotwidth)
 		nr=False
-	
+
 	if options.list[0].exist('ncols'):
 		ncols=options.list[0].getfieldvalue('ncols')
@@ -45,9 +45,9 @@
 	ncols=int(ncols)
 	nrows=int(nrows)
-	
+
 	#check that nrows and ncols were given at the same time!
 	if not nr==nc:
 		raise StandardError('error: nrows and ncols need to be specified together, or not at all')
-	
+
 	#Go through plots
 	if numberofplots:
@@ -82,5 +82,5 @@
 		cbar_size=options.list[0].getfieldvalue('colorbarsize','5%')
 		cbar_pad=options.list[0].getfieldvalue('colorbarpad','2.5%') # None or %
-		
+
 		axgrid=ImageGrid(fig, 111,
 				nrows_ncols=(nrows,ncols),
