Restructure for TSI python package