Update Install script with new functionality