Fix WordCount Example code