Add JS to make checkboxes work