Added getTeam function