Enhance install script