Adds Model testing

Adds tests for all models