Move method to class attribute

To make it easier/cleaner to override in our app.