Refactor setup.sh and setup_util.py

The refactor was necessary to set env vars for
all workspace in one run.

Fixes #289