Added a test for adding relationships. Changed the way relationships are added.