Adding decorator pattern using