Vim folding for python files