Update traits.py