Mostrar si las edades son iguales en python

def main():

#usuario1

edad1=int(input("Ingrese la edad: "))

edad2=int(input("Ingrese la edad: "))

if edad1==edad2:

print("Las edades son iguales")

else:

print("Las edades son diferentes")

if __name__=="__main__":

main()