refactored word_count.js