Update work_space.py