Battleground states look to restore election integrity