Clean python script