Added a bunch of tests for testing models