Make a proper python package to make things easier