Updated call to model class instead of service