Finish rewritin python module