Add checkbox to demo