Small updates to dialog