Adding the message box back