Better handling of install button