Move Dialog classes into ui package