Keras example - using the lambda layer