Adding wizard logic to default ui