Changes to layout of change_ws.py