improved code

Мат. улучшенный код