Add Test for main models