Update space/__init__.py to the latest version
Change 'window_dim' to 'dim' to respect the updated spaces