3 new lines to correct model