Also please rename `n` to `n_samples` to be more explicit and consistent.