Напишите программу , которая в зависимости от номера месяца выведет название времени года в Python

n = int(input())

if (n == 12) or (n == 1) or (n == 2):

   print('Зима')

if (n >= 3) and (n < 6):

   print('Весна')

if (n >= 6) and (n < 9):

   print('Лето')

if (n >= 9) and (n < 12):

   print('Осень')

month = int(input())

if month == 12 or month == 1 or month == 2:

   print("зима")

elif 3 <= month <= 5:

   print("весна")

elif 6 <= month <= 8:

   print("лето")

elif 9 <= month <= 11:

   print("осень")

else:

   print("Неверно введен номер месяца!")