diff --git a/template.rb b/template.rb
index 901906f..692163f 100755
--- a/template.rb
+++ b/template.rb
@@ -1,11 +1,10 @@
 #!/usr/bin/env ruby
 
-problem = :fake
-#input = File.readlines("input", chomp: true)
-input = DATA.read
+input = (ARGV.first.nil? ? DATA : ARGF)
+  .read
 
 result = :solution
 
-puts result
+p result
 
 __END__