change the structure to be a python module