немного подправил код