Add basic code folding