Fancy code folding for python