Add `doc` snippet to python