Adding the code for the decorator pattern