Add random number generation