Clean up on probe function