Update WordCountMain.java

Refactor