Add test for lambdas