Update the schema to include