Added methods to assign and remove roles from an entity