Added lines 3 and 4