wrap this into a python package