TODO: use checkboxes