Add tests for relationships