Sen. Flake says he wants to work with Trump