Allow specifying a function in both the "ast" and "format" commands