Added python equivalent script, doesn't work I don't know why