Небольшие улучшения кода

Небольшие улучшения кода