Tweaks to install script