Functional API to add layers to the Keras Model