trainee task on python from yandex