Improve the install script