Add schema & tests