Building a more complex network in Keras