Adding decorator to lecture 2