Add and use WaitForHealth replicaset functions