Tweaked a code block a little