added back in the rest of the script