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