Doc string clean up