Avoid pass the wait group pointer

[thukabjj]

In the article, Jaroslavs brought some improvements regarding the idiomatic improvement that I could make the first one is the encapsulate the WaitGroup instead of passing the reference by argument.