Updated the sample to the new version

Updated the sample to reflect the changes