自己写的改进版通讯录程序，请大家指教

wfjx_85

原来小甲鱼的答案不够智能，当查询的人不存在时应该直接提醒是否添加。我自己修改了一下，大家看看有什么进一步完善的。
print('|---欢迎进入通讯录程序---|\n|---1:查询联系人资料  ---|\n|---2:插入新的联系人  ---|\n|---3:删除已有联系人  ---|\n|---4:退出通讯录程序  ---|')
mydict = {}
def a():
    name = input('请输入联系人姓名：')
    if name in mydict:
        print(name + ':' + mydict[name])
    else:
        X = input('此联系人不存在，是否添加联系人YES/NO:')
        if X == 'YES':
            tel = input('请输入联系人电话：')
            mydict[name] = tel      
def b():
    name = input('请输入联系人姓名：')
    tel = input('请输入联系人电话：')
    x = 'NO'
    if name in mydict:
        print('您输入的姓名在通讯录中已存在-->>',mydict[name])
        x = input('是否修改用户资料YES/NO')
        if x == 'YES':
            mydict[name] = input('请输入用户联系电话：')
    else:
        mydict[name] = tel
def c():
    name = input('请输入联系人姓名：')
    del mydict[name]
n = 6
while n != 4:
    n = int(input('请输入相关的指令代码：'))
    if n == 1:
       a()
    elif n == 2:
        b()
    elif n == 3:
        c()
    else:
        print('输入错误，请输入1-4的整数')
        n = int(input('请输入相关指令代码：'))
   
print('|---感谢使用通讯录程序 ---|')   
        

wfjx_85

我自己发现了一个问题，n的赋值应该在while语句的外面，否则首次输入4 会提示输入错误，而不是直接退出