Rename to relatedModels and add unit tests