Modifying client for Decorator