Changed lines 4 and 5