Updating code sample to be a bit more accurate