(feat) add associations in schema