Python решить задачу

 def get_age_difference(user_age, program_age=25): 
    age_difference = program_age - user_age 
    if age_difference > 0: 
        return f"Мне {program_age} лет, значит, я старше тебя на {age_difference} лет!" 
    elif age_difference < 0: 
        return f"Мне {program_age} лет, значит, ты старше меня на {-age_difference} лет!" 
    else: 
        return f"Мы одного возраста! Нам по {program_age} лет." 
 
def main(): 
    user_age = input("Сколько тебе лет?: ") 
     
    if user_age.isdigit(): 
        user_age = int(user_age) 
        print(get_age_difference(user_age)) 
    else: 
        print("Пожалуйста, введите корректный возраст.") 
 
if __name__ == "__main__": 
    main() 

 print("Сколько тебе лет?")
age = input()
print(f"А мне {age+2}")