Слегка почистил код