From 6cd76282c02e5fb780e45a926fed40b4aa44e7c8 Mon Sep 17 00:00:00 2001
From: Leen Dereu <leen.dereu@student.kuleuven.be>
Date: Tue, 3 Apr 2018 12:31:22 +0200
Subject: [PATCH] Team class

---
 OGP1718-Worms/src/worms/model/Team.java | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/OGP1718-Worms/src/worms/model/Team.java b/OGP1718-Worms/src/worms/model/Team.java
index 1076f39..5fef8bd 100644
--- a/OGP1718-Worms/src/worms/model/Team.java
+++ b/OGP1718-Worms/src/worms/model/Team.java
@@ -27,14 +27,15 @@ public class Team {
     
     /**
 	 * 
-	 * TODO	Facade:	addWormsToTeam(Team team, Worm... worms)
 	 * 
 	 * @param worm
 	 */
-	public void addWormsToTeam(Worm worm) {
-		if (canHaveAsWorm(worm)) {
-			team.add(worm);
-			sortList(team);
+	public void addWormsToTeam(Worm... worm) {
+		for (Worm w: worm) {
+			if (canHaveAsWorm(w)) {
+				team.add(w);
+				sortList(team);
+			}
 		}
 	}
 	
@@ -61,12 +62,13 @@ public class Team {
 	
 	/**
 	 * 
-	 * TODO Facade:	removeWormsFromTeam(Team team, Worm... worms)
 	 * 
 	 * @param worm
 	 */
-	public void removeWormsFromTeam(Worm worm) {
-		team.remove(worm);
+	public void removeWormsFromTeam(Worm... worm) {
+		for (Worm w: worm) {
+			team.remove(w);
+		}
 	}
 	
 	/**