Minor update to the sample