Add some dialogs to make the logic correct