Make install script more flexible