Update: cleaned the py file