Improved install script