WIP - Add schema