diff --git a/encode_image.rb b/encode_image.rb
index 9e6d7d1..cfe62b3 100644
--- a/encode_image.rb
+++ b/encode_image.rb
@@ -6,13 +6,14 @@ class EncodeImage
 
   @@utils = nil
 
-
   def initialize(p, m, i)
     @@password = p
     @@message  = m
 
     @@utils = ImageUtils.new(@@password, @@message, i)
 
+    @@original_name = n
+
     @@table = @@utils.generateLookupTable
   end
 
@@ -30,23 +31,10 @@ class EncodeImage
       msg_bits = pixel.encodeBits(msg_bits)
     end
 
-#    for x in 0..s[:width]-1
-#      for y in 0..s[:height]-1
-
-#        loc = {"x": x, "y": y}
-#        if @@table.include? loc then
-#          pixel = @@utils.getImage.getPixel(x, y)
-#          msg_bits = pixel.encodeBits(msg_bits)
-#        end
-#      end
-#    end
-
     @@utils.getImage.saveImage
   end
 
-
 # Need to properly sever this at some point...
-
   def decode
 puts "Hacked in decode..."
     s = @@utils.getImage.getSize
@@ -85,4 +73,4 @@ puts "Padding passed scrutiny..."
 
     return msg
   end
-end
\ No newline at end of file
+end
diff --git a/runner.rb b/runner.rb
index 018ccfb..9f2146b 100644
--- a/runner.rb
+++ b/runner.rb
@@ -3,7 +3,9 @@ require "rmagick"
 require "./image_utils.rb"
 require "./encode_image.rb"
 
-img = Magick::ImageList.new "./images/original/arty_2.jpg"
+img_name = "arty"
+
+img = Magick::ImageList.new "./images/original/" + img_name + ".jpg"
 
 #srand(12345)
 
@@ -26,7 +28,7 @@ puts "Initializing EncodeImage..."
 
 start = Time.new.to_f
 
-em = EncodeImage.new(pass, message, img)
+em = EncodeImage.new(pass, message, img, img_name)
 
 puts em.encode
 puts "Took " + (Time.new.to_f - start).to_s + " seconds to encode... 🐱"