Update edit() to use Role model