Tweaked code.