Use traits instead of inheritance for model helper