Подправил код, убрал ненужное