Add test schema and sample code