Add type signature to wordCount