Updates to install wizard