Add schema for validation