diff --git a/README.md b/README.md
index 72d13f3..976de2b 100644
--- a/README.md
+++ b/README.md
@@ -47,6 +47,9 @@ There are two ways to enter filter series
     ```
 
 # Comparison
+
+Image before processing the fingerprint and after it.
+
 <table>
     <thead>
         <th>Before</th>
@@ -58,6 +61,10 @@ There are two ways to enter filter series
     </tbody>
 </table>
 
+# Generated STL model
+
+TODO
+
 # Roadmap
 
 - [x] Load and store image
diff --git a/src/main.py b/src/main.py
index 4114716..2bd666a 100644
--- a/src/main.py
+++ b/src/main.py
@@ -47,8 +47,8 @@ class apply_filters:
         self.height = self.img.shape[0]
         print(self.width, self.height)
         
-        fig = plt.figure(figsize = (self.width/self.dpi, self.height/self.dpi), 
-            frameon = False, dpi = self.dpi)
+        fig = plt.figure(figsize = (self.width, self.height), 
+            frameon = False, dpi = self.dpi / 100) # dpi is in cm
     
         ax = plt.Axes(fig, [0., 0., 1., 1.])
         ax.set_axis_off()