Vertical disc heating in Milky Way-sized galaxies in a cosmological context