improve python module