Creating a neural network using Keras