diff --git a/Assets/Scripts/Models/Friendship.cs b/Assets/Scripts/Models/Friendship.cs
index a5563a9..05daa51 100644
--- a/Assets/Scripts/Models/Friendship.cs
+++ b/Assets/Scripts/Models/Friendship.cs
@@ -14,5 +14,12 @@ namespace DeathBook.Model
 			friend2 = p2;
 			importance = scale;
 		}
+
+		public Person GetFriend(Person p)
+		{
+			if (p == friend1)
+				return friend2;
+			return friend1;
+		}
 	}
 }
diff --git a/Assets/Scripts/Models/LevelGenerator.cs b/Assets/Scripts/Models/LevelGenerator.cs
index 584540e..170eb24 100644
--- a/Assets/Scripts/Models/LevelGenerator.cs
+++ b/Assets/Scripts/Models/LevelGenerator.cs
@@ -135,6 +135,7 @@ namespace DeathBook.Model
 
 			//Person p = new Person(id, pos,);
 			Person p = new Person(id);
+			p.initialPosition = pos;
 
 			return p;
 		}