Move the dialog into fragment to handle rotation + save state of data