Визначити хто из двох учнів вищий та вивести на екран имя(задачи в

nameA, heightA = map(str, input("Введите имя и рост(только число) первого(ой) ученика(цы) через пробел: ").split())

nameB, heightB = map(str, input("Введите имя и рост(только число) второго(ой) ученика(цы) через пробел: ").split())

if heightA > heightB:

   print("Ученик(ца) {0} выше на {1} ед.".format(nameA, int(heightA)-int(heightB)))

elif heightA < heightB:

   print("Ученик(ца) {0} выше на {1} ед.".format(nameB, int(heightB) - int(heightA)))

else:

   print("Рост ученика {0} и {1} равны".format(nameA, nameB))