Update install script to match recent changes