Added model unit tests