Updated doc string to include py 2x