Doc string update