Add code folding

Thanks to Andy Schultz for the suggestion.