Add test case models and migration