Use Python Decorator to enable cluster preservation