add checkbox logic