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