def create_crop(): display_menu() choice = select_option() if choice == 1: new_crop = Potato() elif choice == 2: new_crop = Wheat() return new_crop