Caught between Trump and China, Japan is making a small upgrade to give its submarines an edge