Added support for return. return an expression or a token_id. support for return vector and vector2d.