Python turtle ПОМОГИТЕ ПОЖАЛУЙСТА!! СРОЧНО!!!!!!!!!!!!!!ДАЮ 30 БАЛЛОВ

Користувач вводить назву фігури: "Квадрат, Прямокутник або Трикутник". Залежно від введення користувача за допомогою черепашки малюється дана фігура.

import turtle

def draw_square():

   for _ in range(4):

       turtle.forward(100)

       turtle.left(90)

def draw_rectangle():

   for _ in range(2):

       turtle.forward(100)

       turtle.left(90)

       turtle.forward(50)

       turtle.left(90)

def draw_triangle():

   for _ in range(3):

       turtle.forward(100)

       turtle.left(120)

def draw_figure():

   figure = turtle.textinput("Фігура", "Введіть назву фігури: ")

   if figure == "Квадрат":

       draw_square()

   elif figure == "Прямокутник":

       draw_rectangle()

   elif figure == "Трикутник":

       draw_triangle()

   else:

       print("Невірно введена назва фігури!")

       turtle.bye()

draw_figure()