I can add the method to the entity.. that's a good idea!