fix(script): Convert script to python