diff --git a/.gitignore b/.gitignore
index 8b9b63a..2747e1e 100644
--- a/.gitignore
+++ b/.gitignore
@@ -5,4 +5,6 @@ Thumbs.db
 *.war
 *.ear
 *.log
-*.so
\ No newline at end of file
+*.so
+*.jpg
+*.png
\ No newline at end of file
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index 48c5fc3..856a488 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -2,8 +2,11 @@
 <project version="4">
   <component name="ChangeListManager">
     <list default="true" id="37dc1861-899d-4bf2-8d5f-5f15145ed4ca" name="Default Changelist" comment="">
+      <change beforePath="$PROJECT_DIR$/.gitignore" beforeDir="false" afterPath="$PROJECT_DIR$/.gitignore" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/out/production/BookApp/com/github/book_folding_app/BookFolder.class" beforeDir="false" afterPath="$PROJECT_DIR$/out/production/BookApp/com/github/book_folding_app/BookFolder.class" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/src/com/github/book_folding_app/BookFolder.java" beforeDir="false" afterPath="$PROJECT_DIR$/src/com/github/book_folding_app/BookFolder.java" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/test_out.png" beforeDir="false" />
     </list>
     <ignored path="$PROJECT_DIR$/out/" />
     <option name="EXCLUDED_CONVERTED_TO_IGNORED" value="true" />
@@ -17,8 +20,8 @@
       <file pinned="false" current-in-tab="true">
         <entry file="file://$PROJECT_DIR$/src/com/github/book_folding_app/BookFolder.java">
           <provider selected="true" editor-type-id="text-editor">
-            <state relative-caret-position="426">
-              <caret line="195" column="59" lean-forward="true" selection-start-line="195" selection-start-column="59" selection-end-line="195" selection-end-column="59" />
+            <state relative-caret-position="120">
+              <caret line="179" column="61" selection-start-line="179" selection-start-column="61" selection-end-line="179" selection-end-column="61" />
               <folding>
                 <element signature="imports" expanded="true" />
                 <element signature="method#get_str#0;class#BookFolder#0" />
@@ -28,8 +31,8 @@
                 <element signature="method#get_page_cuts#0;class#BookFolder#0" />
                 <element signature="method#check_page_cuts#0;class#BookFolder#0" />
                 <element signature="method#get_img#0;class#BookFolder#0" />
-                <element signature="method#get_img_gray#0;class#BookFolder#0" />
                 <element signature="method#resize_img#0;class#BookFolder#0" />
+                <element signature="method#get_img_gray#0;class#BookFolder#0" />
                 <element signature="method#run#0;class#BookFolder#0" />
               </folding>
             </state>
@@ -197,8 +200,8 @@
     <entry file="file://$PROJECT_DIR$/src/bookFold.java" />
     <entry file="file://$PROJECT_DIR$/src/com/github/book_folding_app/BookFolder.java">
       <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="426">
-          <caret line="195" column="59" lean-forward="true" selection-start-line="195" selection-start-column="59" selection-end-line="195" selection-end-column="59" />
+        <state relative-caret-position="120">
+          <caret line="179" column="61" selection-start-line="179" selection-start-column="61" selection-end-line="179" selection-end-column="61" />
           <folding>
             <element signature="imports" expanded="true" />
             <element signature="method#get_str#0;class#BookFolder#0" />
@@ -208,8 +211,8 @@
             <element signature="method#get_page_cuts#0;class#BookFolder#0" />
             <element signature="method#check_page_cuts#0;class#BookFolder#0" />
             <element signature="method#get_img#0;class#BookFolder#0" />
-            <element signature="method#get_img_gray#0;class#BookFolder#0" />
             <element signature="method#resize_img#0;class#BookFolder#0" />
+            <element signature="method#get_img_gray#0;class#BookFolder#0" />
             <element signature="method#run#0;class#BookFolder#0" />
           </folding>
         </state>
diff --git a/out/production/BookApp/com/github/book_folding_app/BookFolder.class b/out/production/BookApp/com/github/book_folding_app/BookFolder.class
index d97df50..2e7cdcc 100644
Binary files a/out/production/BookApp/com/github/book_folding_app/BookFolder.class and b/out/production/BookApp/com/github/book_folding_app/BookFolder.class differ
diff --git a/src/com/github/book_folding_app/BookFolder.java b/src/com/github/book_folding_app/BookFolder.java
index 4f8cdde..1ee871a 100644
--- a/src/com/github/book_folding_app/BookFolder.java
+++ b/src/com/github/book_folding_app/BookFolder.java
@@ -128,13 +128,8 @@ private float get_height(float num, float px_height, float height) {
         return num / px_height * height;
     }
 
-    // Convert height from inches to pixels
-    private int get_px_height(float num, float px_height, float height) {
-        return (int)(num / height * px_height);
-    }
-
     // Retrieve specified image
-    BufferedImage get_img() {
+    private BufferedImage get_img() {
         BufferedImage img = null;
         try {
             img = ImageIO.read(new File(img_name));
@@ -145,8 +140,18 @@ BufferedImage get_img() {
         return img;
     }
 
+    // Resize image
+    private BufferedImage resize_img(BufferedImage img) {
+        Image img_temp = img.getScaledInstance(num_pages, px_height, Image.SCALE_SMOOTH);
+        BufferedImage img_resize = new BufferedImage(num_pages, px_height, BufferedImage.TYPE_BYTE_GRAY);
+        Graphics2D graphics = img_resize.createGraphics();
+        graphics.drawImage(img_temp, 0, 0, null);
+        graphics.dispose();
+        return img_resize;
+    }
+
     // Convert image to grayscale
-    BufferedImage get_img_gray(BufferedImage img) {
+    private BufferedImage get_img_gray(BufferedImage img) {
         BufferedImage img_grayscale = new BufferedImage(
                 img.getWidth(),
                 img.getHeight(),
@@ -157,50 +162,42 @@ BufferedImage get_img_gray(BufferedImage img) {
         return img_grayscale;
     }
 
-    // Resize image
-    BufferedImage resize_img(BufferedImage img) {
-        BufferedImage img_resize = new BufferedImage(num_pages, (int)height, BufferedImage.TYPE_INT_RGB);
-        return img_resize;
-    }
-
     // Convert grayscale image to 2D array of pixels
 
-    // Save image
+    // Perform image manipulation functions
+    private BufferedImage edit_img() {
+        BufferedImage img = get_img();
+        BufferedImage img_resize = resize_img(img);
+        BufferedImage img_gray = get_img_gray(img_resize);
+        return img_gray;
+    }
 
+    // Save image
+    private void save_img(BufferedImage img, String file_path, String file_name) {
+        File out_file = new File(file_path + File.separator + file_name);
+        System.out.println(out_file);
+        try {
+            ImageIO.write(img, img_name.substring(img_name.lastIndexOf('.') + 1), out_file);
+        }
+        catch (IOException e) {
+            System.out.println("ERROR: File not saved");
+        }
+    }
 
     private void test() {
+        num_pages = 100;
+        height = 10f;
+        px_height = (int)(height * 96);
         img_name = get_str("\nEnter image name\n");
-        BufferedImage img = get_img();
+        BufferedImage img = edit_img();
         // Try using this:
         // C:\Users\g3bry\Desktop\Python\BookProject\pugtest.png
-
-        // This works as long as the image is placed in the \BookApp dir
+        // Works as long as the image is placed in the \BookApp dir
         String working_dir = System.getProperty("user.dir");
         String abs_file_path = working_dir + File.separator + img_name;
         System.out.println(abs_file_path);
 
-        // Try resizing
-        num_pages = 100;
-        height = 10f;
-        px_height = (int)(height * 96);
-        Image img_temp = img.getScaledInstance(num_pages, px_height, Image.SCALE_SMOOTH);
-        BufferedImage img_resize = new BufferedImage(num_pages, px_height, BufferedImage.TYPE_BYTE_GRAY);
-        Graphics2D graphics = img_resize.createGraphics();
-        graphics.drawImage(img_temp, 0, 0, null);
-        graphics.dispose();
-
-        // This works
-        BufferedImage img_gray = get_img_gray(img_resize);
-
-        // Write manipulated file
-        File out_file = new File(working_dir + File.separator + "test_out.png");
-        try {
-            ImageIO.write(img_resize, "png", out_file);
-        }
-        catch (IOException e) {
-            System.out.println("Error");
-        }
-
+        save_img(img, working_dir, "test_out" + img_name.substring(img_name.length() - 4));
     }
 
     private void run() {