This setup stops due pymatgen dependency (a net improvement)