Add counter assertions and WordCount test