Updates to install script