Adsorption and diffusion of lithium on heteroatom-doped monolayer molybdenum disulfide