I suggest using ```functools.wraps``` here.