code corrected