Add wrapper.py, and test_wrapper.py