Немного почистил код