Немного улучшил код