Update to use script