Python Wrap A Function