Organize into python package