complete 'set_up_teams' method in source.py