diff --git a/DiffractionGratings/grating3D_postplot.py b/DiffractionGratings/grating3D_postplot.py
index 3abf0c89ee825e5ab8f5913be5e72b330b225d30..2c3b472e801acd7984fb1cd15b492a6a92715755 100644
--- a/DiffractionGratings/grating3D_postplot.py
+++ b/DiffractionGratings/grating3D_postplot.py
@@ -2,17 +2,12 @@ import numpy as np
 import matplotlib.pyplot as pl
 import sys
 myDir = sys.argv[1]
-# myDir = res3D
 Rnm = np.loadtxt(myDir+'/eff_r.txt',ndmin=2)[:,1] + 1j*np.loadtxt(myDir+'/eff_r.txt',ndmin=2)[:,2]
 Tnm = np.loadtxt(myDir+'/eff_t.txt',ndmin=2)[:,1] + 1j*np.loadtxt(myDir+'/eff_t.txt',ndmin=2)[:,2]
 Q = [np.loadtxt(myDir+'/temp-Q_L_%g.txt'%k,ndmin=2)[:,1] for k in range(2,7)]
 Q.append(np.loadtxt(myDir+'/temp-Q_scat.txt',ndmin=2)[:,1])
 Q=np.array(Q)
 TOT = Rnm.real.sum()+Tnm.real.sum()+Q.sum()
-print('Rtot',Rnm.real.sum())
-print('Ttot',Tnm.real.sum())
-print('Atot',Q.sum())
-print('TOT ',TOT)
 
 if myDir[6:]=='solarcell':
     Nmax=2
@@ -25,13 +20,19 @@ if myDir[6:]=='solarcell':
     Abs_rods = Q[-1]
     Abs_ITO  = Q[0]
     Abs_subs  = Q[2]+Q[3]+Q[4]+Ttot
+    TOT = Rtot+Abs_rods+Abs_ITO+Abs_subs
     pl.figure()
     pl.plot(tab_lambdas,Abs_ITO,label='absorption ITO electrode')
     pl.plot(tab_lambdas,Abs_rods,label='absorption in Si rods')
     pl.plot(tab_lambdas,Abs_subs,label='absorption in Si subs')
     pl.plot(tab_lambdas,Rtot,label='reflection')
+    pl.plot(tab_lambdas,TOT,label='total')
     pl.legend()
-    pl.xlabel('$\lambda$ [nm]')
+    pl.xlabel(r'$\lambda$ [nm]')
     pl.ylabel('fraction of incident energy')
     pl.savefig('solar_balance.pdf')
-    pl.show()
\ No newline at end of file
+else:
+    print('Rtot',Rnm.real.sum())
+    print('Ttot',Tnm.real.sum())
+    print('Atot',Q.sum())
+    print('TOT ',TOT)