Small improvements for word_count.clj sample.