move wg.Done back to sim.go