Make this a proper python package