kolor = 'żółty'

if kolor == 'czerwony':
    print('Zatrzymaj się!')
elif kolor == 'żółty':
    print('Przygotuj się!')
else:
    print('Jedź!')