Improved package structure

README.md

@@ -30,7 +30,7 @@ Currently the code supports the following parameters:
 Creating machine learning models in Keras is quite easy, but I think it can be easier. Editing layers, adding layers, removing layers, it all requires coding. Coding takes time, and code has to be debugged. That is why I created this tool, were the structure of a machine learning model is merely configuration, a set of variables in a file. It also makes it easier to create machine learning models from other code, and to use libraries different than Keras if wanted.
 
 ## But Keras has a model format already
-Yes, and you can do more with it than with this tool. The API of this tool is more important than the json format, which is still a reason to create this tool.
+Yes, and you can do more with it than with this tool. The API of this tool is more important than the json format, which is still a reason to create this tool. The ModelBuilder also tries to help you, making the creation of models easier. Input, Activation and Reshape layers are automatically created where necessary.
 
 ## Requirements
 * Python >= 3.6

model_builder/__init__.py

@@ -0,0 +1,5 @@
+from model_builder.model_builder import ModelBuilder
+from model_builder import model_analyzer
+from model_builder import model_parser
+from model_builder.model_serializer import ModelSerializer
+from model_builder.model import Model

setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name='model_builder',
-    version='0.1.0',
+    version='0.1.1',
     author="Wilmer Zwietering",
     author_email="[email protected]",
     description="Easily build Keras models from code or from a configuration file",