add @wraps decorator