Add in more logic to determine type of install