Add decorator for token management