Update to use DialogFragment