Make into proper python package