Updated variables to match context of problem
Got tests to pass.
Refactored code to create #get_words
Simplified #word_count