Add folding functions for expressions and arguments