Add more lambda test code