My code optimization

Старался