Make the doc string similar to how we do it for python