Updating doc for SampleByKey