Simpler solution without button as models.