diff --git a/bin/save_model b/bin/save_model
index 37362cc..10c6b81 100755
--- a/bin/save_model
+++ b/bin/save_model
@@ -2,10 +2,10 @@
 require 'date'
 
 models_dir = 'models'
-model_name = ARGV[0]
+model_name = (ARGV[0] or "default")
 model_path = File.join(models_dir, model_name)
 
-episode_count = (File.read File.join(model_path, 'model.episodes')).to_i
+episode_count = (File.read File.join(model_path, 'episodes_trained')).to_i
 
 puts "Found model #{model_name} with episodes #{episode_count} trained!"