corrected

lines 16 and 20