Wrote model tests