parse_schema.py now returns a dictionary