Murray mentally switched on for US Open