Trying to model the schema and test it with mock data