Add doc string for python script