Code review
Merge pull request #1 from Flamdoodle/model_testing

Code review