Move logic to view subclass