Clean up doc string