Wrote a few tests for Models