Adding additional wizard functionality