feat: improve function expansion
feat: improve for expansion
feat: improve if expansion
feat: expand f to =>