Add related data tests