Update trait_types.py