подправил код