Use stop channel and waitgroup