Update workstation.py