Generalize atomtype.py to prepare for additional elements