Add relationship tests