Working on ui/Dialog