include model class

src/datastore.py

src/model_modules/model_class.py

src/modules/experiment_setup.py

src/modules/model_setup.py

src/modules/pre_processing.py

test/test_datastore.py

test/test_model_modules/test_model_class.py

test/test_modules/test_model_setup.py

test/test_modules/test_pre_processing.py

test/test_modules/test_training.py