Added model tests