Attempting to add decorator