use .. code:: python in documentation