Minor code refactoring
Simplify flatten code
Add Expression.args_str method