Add test for roles and modify migrations