That wasn't the only reason they scrapped it. There is a good argument to be made that the 3.5mm was a redundant port because lightning/USB C are both capable of sending audio (with the assistance of an adapter until lightning/USB C headphones are more common).