[WIP] Add save function to workspace