Add probe sampling function