readme code snippet updated