Simplify expression type

Replace lambda function arguments with concrete representation
Add rudimentary support for model extraction