Wal-Mart to sell only U.S. shrimp in 3 states