Upgrade to pymatgen 3.0.6