Add argument to set random seed.