Add a Decorator Pattern and an unit test for it.