Fixed bug: wait group now utilized properly.