Also this can be more straightforwardly done with subclassing