added roles to model and tested