Mental coach helped Stewart focus ahead of U.S. Open win