Add wait group to calculate full time