Help with KeyError: 0 in Material Assignment Using FenicsX