Add schema.py back