Added the ability to add checkboxes instead of icons