Update Team class and tests