Iteration 2 on wordcount task