Added writing roles into the tests