Modify create role function