Adding in random num generation