Update to install script