Additional modules in Keras