Improve install script