Adding Decorator Pattern