added in dialog also