make it more python package