Tweaked the code script.