Baby formula supply improving, research firm says