Fix segue from custom widget demo
Updates