Added random number generator to test