diff --git a/adaptiveumbrella/__init__.py b/adaptiveumbrella/__init__.py
index ffb7e11..f4e1178 100644
--- a/adaptiveumbrella/__init__.py
+++ b/adaptiveumbrella/__init__.py
@@ -3,4 +3,4 @@ from .runner import UmbrellaRunner
 from .wham2d import WHAM2DRunner
 
 __all__ = ['AdaptiveUmbrella']
-__version__ = "0.3.1"
+__version__ = "0.3.2"
diff --git a/adaptiveumbrella/runner.py b/adaptiveumbrella/runner.py
index 8b94d25..18e4171 100755
--- a/adaptiveumbrella/runner.py
+++ b/adaptiveumbrella/runner.py
@@ -193,14 +193,15 @@ class UmbrellaRunner():
 
             print("Running simulations")
             self.simulate_frames(lambdas, new_frames)
-
-            print("Calculating new PMF")
-            self.pmf = self.calculate_new_pmf()
-
+    
             # update list of sampled windows
             for new_frame in new_frames.keys():
                 self.sample_list[new_frame] = self.num_iterations
 
+
+            print("Calculating new PMF")
+            self.pmf = self.calculate_new_pmf()
+
             self.after_run_hook()
 
             if(self.max_iterations > 0 and self.num_iterations == self.max_iterations):
diff --git a/adaptiveumbrella/wham2d.py b/adaptiveumbrella/wham2d.py
index a67dc2d..03823e6 100644
--- a/adaptiveumbrella/wham2d.py
+++ b/adaptiveumbrella/wham2d.py
@@ -33,7 +33,9 @@ class WHAM2DRunner(UmbrellaRunner):
         path = os.path.join(self.tmp_folder, "{}_metadata.dat".format(self.num_iterations))
         with open(path, 'w') as out:
             for file in os.listdir(self.simulation_folder):
-                if not os.path.exists(os.path.join(path, file, "COLVAR")):
+                filepath = os.path.join(self.simulation_folder, file, "COLVAR")
+                if not os.path.exists(filepath):
+                    print("Not found: {}".format(filepath))
                     continue
                 prefix, x, y = file.split("_")
                 colvar_file = os.path.join(file, 'COLVAR')
@@ -47,18 +49,38 @@ class WHAM2DRunner(UmbrellaRunner):
         """ Output file for wham-2d """
         return os.path.join(self.tmp_folder, 'freeenergy_tmp.dat')
 
+    def get_wham_borders(self):
+        """ we need to find dimensions for wham-2d calculations
+        that are as small as possible but contain all sampled frames. """
+        nonzero = np.nonzero(self.sample_list)
+        borders = np.array(
+            self._get_lambdas_for_index((nonzero[0].min(), nonzero[1].min())),
+            self._get_lambdas_for_index((nonzero[0].max(), nonzero[1].max())))
+        
+        borders = borders.flatten()
+        
+        # increase borders by 1 lambda step from minimal dimensions                        
+        borders[0] -= self.cvs[0][2]
+        borders[1] -= self.cvs[1][2]
+        borders[2] += self.cvs[0][2]
+        borders[3] += self.cvs[1][2]
+        return flat
+
+
     def run_wham2d(self, metafile_path, output_path):
         """ Runs wham-2d with the given parameters. See http://membrane.urmc.rochester.edu/sites/default/files/wham/doc.html """
 
+
+        borders = self.get_wham_borders()
         cmd = "{exec} Px={px} {min_x} {max_x} {frames_x} Py={py} {min_y} {max_y} {frames_y} {tol} 298 0 {metafile} {outfile} 0".format(
             exec=self.WHAM_EXEC,
             px=self.whamconfig['Px'],
-            min_x=self.whamconfig['hist_min_x'],
-            max_x=self.whamconfig['hist_max_x'],
+            min_x=borders[0],
+            max_x=borders[2],
             frames_x=self.whamconfig['num_bins_x'],
             py=self.whamconfig['Py'],
-            min_y=self.whamconfig['hist_min_y'],
-            max_y=self.whamconfig['hist_max_y'],
+            min_y=borders[1],
+            max_y=borders[3],
             frames_y=self.whamconfig['num_bins_y'],
             tol=self.whamconfig['tolerance'],
             metafile=metafile_path,
diff --git a/examples/data/sim.sh b/examples/data/sim.sh
index a61450b..f36231a 100755
--- a/examples/data/sim.sh
+++ b/examples/data/sim.sh
@@ -17,7 +17,8 @@ sed -i -e "s/LAMBDA1/${lambda1}/g" $FOLDER/plumed.dat
 sed -i -e "s/LAMBDA2/${lambda2}/g" $FOLDER/plumed.dat
 
 cd $FOLDER
-source /usr/local/gromacs/bin/GMXRC
-gmx --quiet mdrun -deffnm topol -plumed plumed.dat -nsteps 250000 2>&1
+#source /usr/local/gromacs/bin/GMXRC
+source /opt/gromacs-2018/bin/GMXRC
+gmx --quiet mdrun -deffnm topol -plumed plumed.dat -nsteps 100000 2>&1
 cd ../..
 
diff --git a/examples/example.py b/examples/example.py
index 86a78a3..c96fbe4 100755
--- a/examples/example.py
+++ b/examples/example.py
@@ -98,6 +98,6 @@ runner.cvs_init = (0, 0)
 runner.E_min = 5
 runner.E_max = 100
 runner.E_incr = 10
-runner.max_iterations = 5
+runner.max_iterations = 1
 
 runner.run()