Add count-words-partitioned, courtesy of @stuarthalloway