Updates to wizard