lib/femis_hangman/word.rb in femis_hangman-0.1.1 vs lib/femis_hangman/word.rb in femis_hangman-0.1.2

@@ -3,14 +3,12 @@
     def generate(difficulty)
       file = File.open('./dictionary.txt', 'r')
       rand(41211).times { file.gets }
       word = clean_word($_)
       file.close
-      if confirm(difficulty, word)
-        word
-      else
-        generate(difficulty)
+      if confirm(difficulty, word) then word
+      else generate(difficulty)
       end
     end
 
     def confirm(difficulty, word)
       if word.length > 4 * difficulty && word.length <= 4 * (difficulty + 1)
@@ -19,9 +17,9 @@
         false
       end
     end
 
     def clean_word(word)
-      cleaned_word = word.gsub("\n", '')
+      word.delete("\n")
     end
   end
 end
\ No newline at end of file