diff --git a/crpptbx/AUG/torbeam_prepare_inputs_and_run.m b/crpptbx/AUG/torbeam_prepare_inputs_and_run.m
index dc419e20075b3933a75f49cbb4f6aac16e109af4..2d908fc727932420407c7cc16d01ec794e5d309e 100644
--- a/crpptbx/AUG/torbeam_prepare_inputs_and_run.m
+++ b/crpptbx/AUG/torbeam_prepare_inputs_and_run.m
@@ -212,7 +212,7 @@ for ibeam=find(any(torbeam_out.run_torbeam,2))'
   figure;
   subplot(2,1,1)
   plot(squeeze(torbeam_out.pdens_jcd_rho(:,ibeam,:)),squeeze(torbeam_out.pdens(:,ibeam,:)))
-  title(['for beam = ' num2str(ibeam)])
+  title(['AUG shot #' num2str(shot) ': for beam = ' num2str(ibeam)])
   ylabel('Pdens')
   xlabel('r/a')
   subplot(2,1,2)
@@ -226,6 +226,8 @@ subplot(3,1,1)
 plot(torbeam_out.t,torbeam_out.pabs/1e6,'*-')
 ylabel('Pabs [MW]')
 
+title(['AUG shot #' num2str(shot)])
+
 subplot(3,1,2)
 plot(torbeam_out.t,torbeam_out.icd/1e6,'*-')
 ylabel('Icd [MA]')