Added @functools.wraps for travis decorator