add word count code