"Why doesn't the US instead work more with Mexico to improve conditions in that country?"
That was what NAFTA was about. If Trump terminates NAFTA and brings jobs "back home", more Mexicans will want to enter the U.S.