added team class and changed tests