Can you rename `rand_generator` to `random_state` for consistency with the rest of the codebase?