Add function code folding