The remove dialog is now working as the other dialog with its own function in the message handler.