Added simple assignRoles logic