Adds getTeam function