Correcting doc string