added message box, but have a bug...