removed pre-existing unused imports
made TensorStorageVariable be allowed to be an update target
updated doc of how storage works with cpu and gpu