Update: Add schema and callbacks