Update schema.py