Example of batch normalization with Keras