[TASK] Update UI to role models