Neural Program Search with Higher-Order Functions and Lambdas