Updaes to wizard