Add replica set initialization