reviewed code corrected + PEP8