diff --git a/kalman/graphics/figure2.py b/kalman/graphics/figure2.py
index 8b71483..4ac9617 100644
--- a/kalman/graphics/figure2.py
+++ b/kalman/graphics/figure2.py
@@ -56,15 +56,20 @@ def plot_parameters(dat, input_file, deparameterize=False, ref=None):
   
     
     current_val = []
+    labels = []
     
     for bnd_c in inputfile['estimation']['boundary_conditions']:
+        
         if 'windkessel' in bnd_c['type']:
             for bnd_set in inputfile['boundary_conditions']:
                 if bnd_c['id'] == bnd_set['id']:
                     current_val.append(bnd_set['value'][0])
+                    labels.append('R_' + str(bnd_c['id']))
 
+        elif 'dirichlet' in bnd_c['type']:
+            current_val.append(inputfile['boundary_conditions'][1]['parameters']['U'])
+            labels.append('U')
 
-    current_val.append(inputfile['boundary_conditions'][1]['parameters']['U'])
 
     dim = dat['theta'].shape[-1]
     fig1, axes = plt.subplots(1,1,figsize=(8,6))
@@ -79,8 +84,8 @@ def plot_parameters(dat, input_file, deparameterize=False, ref=None):
 
     col = cycle(['C0', 'C1', 'C2', 'C3','C4'])
     ls = cycle(['-', '-', '--', '--', ':', ':', '-.', '-.'])
-    legends = cycle(['$R_3$','$R_4$','$R_5$','$R_6$','$U$'])
-    #legends = cycle(['$R_3$','$U$'])
+    #legends = cycle(['$R_3$','$R_4$','$R_5$','$R_6$','$U$'])
+    legends = cycle(labels)
     
 
     col_ = next(col)
diff --git a/kalman/input_files/aorta_C.yaml b/kalman/input_files/aorta_C.yaml
index 71b5913..ad9fabb 100755
--- a/kalman/input_files/aorta_C.yaml
+++ b/kalman/input_files/aorta_C.yaml
@@ -6,7 +6,7 @@ fluid:
     dynamic_viscosity: 0.035
 
 io:
-    write_path: 'results/aorta_C'
+    write_path: 'results/aorta'
     restart:
         path: ''  # './projects/nse_coa3d/results/test_restart2/' 
         time: 0
@@ -33,36 +33,36 @@ boundary_conditions:
                     - U*sin(DOLFIN_PI*(t-4*Tc)/Th)*(t>4*Tc)*(t<= 4*Tc + Th)
                     + (t<=5*Tc)*(t>4*Tc+Th)*(-3.67949466208*U*sin(9*DOLFIN_PI*(t-4*Tc)/Th)*exp(-(t-4*Tc)*10)) ']
         parameters:
-            U: 100  #100 safe
+            U: 100  #100
             Th: 0.35
             Tc: 0.8
             t: 0   
     - 
         id: 3
         type: 'windkessel'
+        #value: [10,0,0]
         value: [10,0.0008,2400]
-        #value: [10,0,2400]
         p0: [80,1333.223874]
         #p0: [0,1333.223874]
     - 
         id: 4
         type: 'windkessel'
-        value: [60,0.00034,4200]
-        #value: [60,0,4200] #safe
+        value: [60,0,0]
+        #value: [60,0.00034,4200]
         p0: [80,1333.223874]
         #p0: [0,1333.223874]
     - 
         id: 5
         type: 'windkessel'
-        value: [220,0.00034,11000]
-        #value: [220,0,11000] #safe
+        value: [220,0,0]
+        #value: [220,0.00034,11000]
         p0: [80,1333.223874]
         #p0: [0,1333.223874]
     - 
         id: 6
         type: 'windkessel'
-        value: [160,0.00034,7800]
-        #value: [160,0,7800] #safe
+        value: [160,0,0]
+        #value: [160,0.00034,7800]
         p0: [80,1333.223874]
         #p0: [0,1333.223874]
 
@@ -144,20 +144,21 @@ estimation:
         #    id: 6
         #    type: 'windkessel'
         #    initial_stddev: 1
-        - 
-            id: 2
-            type: 'dirichlet'
-            parameters: 'U'
-            initial_stddev: 1
+        #- 
+        #    id: 2
+        #    type: 'dirichlet'
+        #    parameters: 'U'
+        #    initial_stddev: 1
 
     measurements:
         -
             mesh: './meshes/coaortaH3_leo2.0.h5'
+            #mesh: './meshes/coaortaH1.h5'
             fe_degree: 1
             #xdmf_file: 'measurements/aorta_C/Perturbation/Ks12V50/u_all.xdmf'
             #file_root: 'measurements/aorta_C/Perturbation/Ks12V50/u{i}.h5'
-            xdmf_file: 'measurements/aorta_C_s100/u_all.xdmf'
-            file_root: 'measurements/aorta_C_s100/u{i}.h5'
+            xdmf_file: 'measurements/aorta_s100/u_all.xdmf'
+            file_root: 'measurements/aorta_s100/u{i}.h5'
             indices: 0 # indices of checkpoints to be processed. 0 == all
             velocity_direction: ~
             noise_stddev: 15     # standard deviation of Gaussian noise