Updating the code snippet to be the full thing.