Add test cases for relations and new model methods