Add decorator pattern
add decorator pattern