Tweaked the code snippet.