First try at implementing Team class and rand_team function