Немного изменил код