Add code folding capabilities