word-count: Adopt functional style

As discussed in #79.