Updated to reflect changes to script.