Add code folding for python